import pandas as pd

# 创建测试数据
data = {
    '姓名': ['张三', '李四', '王五', '赵六'],
    '年龄': [25, 30, 35, 40],
    '城市': ['北京', '上海', '广州', '深圳'],
    '薪资': [15000, 20000, 25000, 30000]
}

# 创建DataFrame
df = pd.DataFrame(data)

# 基本操作示例
print("原始数据:")
print(df)

print("\n前两行数据:")
print(df.head(2))

print("\n描述性统计:")
print(df.describe())

print("\n按薪资排序:")
print(df.sort_values('薪资', ascending=False))

print("\n北京地区人员:")
print(df[df['城市'] == '北京'])

# 添加新列
df['薪资等级'] = df['薪资'].apply(lambda x: '高' if x > 20000 else '中')
print("\n添加薪资等级列后:")
print(df)